# 作业要求：*优化课程表

# 导入模块
import datetime
time = datetime.datetime.today().weekday()
weekday = int(time)

# 模拟数据库
username = user_data = ["2010", "2012", "2014"]
password = password_data = ["111111"]
vip_user_data = ['2010']
vip_password_data = ["111111"]

# 输入次数（只有五次登陆机会）
count = 5
while True:
    count -= 1

    # 先判断该账号是否在本网站注册
    if username in user_data or username in vip_user_data:

        # 如果用户为VIP账户
        if username in vip_user_data:
            count = 10  # 代表VIP用户有10次查询课表的机会
    # 如果用户为普通账户
        elif username in user_data and password in password_data:
            count = 5  # 代表普通用户有5次查询课表的机会


# 用户登陆
    username = input("输入您的用户名：")
    password = input("输入您的用户密码：")

    # 进入课程表
    print("Welcome to", username, "to use this website")
    print("You have ", count, " chances")

# VIP用户查询
    if username in vip_user_data and password in vip_password_data:
        count = 10
        while True:
            count -= 1
            if count == -1:
                print("抱歉，您的查询次数已经用完，请明天登录再试")
                break
            else:
                print("亲爱的用户，您今天还有，count，次机会")
                weekday = int(input('请输入您要查询的日期'))

            if weekday == 1:
                print("今天有两门课程,4-5节 大英（10-204）,9-11节 偶练歌舞（音乐楼）快点滚去上课吧")
            if weekday == 2:
                print("今天有两门课程,3-5节 创业实践（1-205）,8-10节 网站运营与管理（3实-304）快点滚去上课吧")
            if weekday == 3:
                print("今天有两门课程,1-3节 毛概理论（7-104）,12-13节 大英（10-204）快点滚去上课吧")
            if weekday == 4:
                print("今天有两门课程,3-5节 python（新综-612）,8-9节 毛概实践（7-307）快点滚去上课吧")
            if weekday == 5:
                print("今天只有一门课程,3-5节 瑜伽（东体-武术馆）快点滚去上课吧")
        else:
            print("难得周末,终于可以好好休息~享受周末的时间吧")


# 以下为普通用户查询
    if username in user_data and password in password_data:
        count = 5
    while True:
        count -= 1
        if count == -1:
            print("对不起，您今天的查询机会已经用完，请明天再登录系统查询")
            break

        else:
            print("亲爱的用户，您今天还有，count，次机会")
            weekday = int(input('请输入您要查询的日期'))

    if weekday == 1:
        print("今天有两门课程,4-5节 大英（10-204）,9-11节 偶练歌舞（音乐楼）快点滚去上课吧")
    if weekday == 2:
        print("今天有两门课程,3-5节 创业实践（1-205）,8-10节 网站运营与管理（3实-304）快点滚去上课吧")
    if weekday == 3:
        print("今天有两门课程,1-3节 毛概理论（7-104）,12-13节 大英（10-204）快点滚去上课吧")
    if weekday == 4:
        print("今天有两门课程,3-5节 python（新综-612）,8-9节 毛概实践（7-307）快点滚去上课吧")
    if weekday == 5:
        print("今天只有一门课程,3-5节 瑜伽（东体-武术馆）快点滚去上课吧")
    else:
        print("难得周末,终于可以好好休息~享受周末的时间吧")
